package zisu.netty.practice.netty.groupchat;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;

/**
 * @Desc: ""
 * @Author: caixiang
 * @DATE: 2020/7/17 9:09
 */
public class GroupChatServerHandler extends SimpleChannelInboundHandler<String> {

    //定义一个channel组 ，管理所有的channel;;GlobalEventExecutor.INSTANCE是全局事件执行器，是一个单例
    //ChannelGroup 就相当于一个channel 的List，，对ChannelGroup操作就相当于对所有这个group里的channel操作这样是比较方便的
    private static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
    SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH-mm-ss");

    //todo 也可以通过HashMap<Channel> 来实现私聊
    public static HashMap<String,Channel> channelHashMap = new HashMap<>();

    //handlerAdded 表示连接建立，一旦连接，这方法是第一个被执行的方法
    //将 当前channel 加入到channelGroup 里面去
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        //将该客户加入聊天 的信息推送给其他在线的的客户端。
        //channelGroup.writeAndFlush 可以对这个group的所有channel write
        channelGroup.writeAndFlush("[客户端] :"+channel.remoteAddress()+"加入聊天"+ sdf.format(new Date()) +" \n");
        channelGroup.add(channel);

        //私聊的
        channelHashMap.put("id100",channel);
    }

    //handler断开连接会被触发,,当xx客户离开信息推送给当前所有在线客户
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        //当handlerRemoved执行的时候，netty 会自动的把这个channel从channelGroup里面移除。
        channelGroup.writeAndFlush("[客户端]"+ctx.channel().remoteAddress()+"离开了"+ sdf.format(new Date()) +" \n");
        System.out.println("channelGroup size is :"+channelGroup.size());
    }

    //下面两个方法 和 上面两个方法并不冲突
    //表示channel 处于一个活动状态，，提示xxx上线
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress()+"上线了");
    }
    //表示channel 处于不活动状态，，提示xxx下线
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress()+"离线了");
    }

    //读取数据
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        //获取当前channel
        Channel channel = ctx.channel();

        //这时候遍历channelGroup ，根据不同的情况，会送不同的消息
        channelGroup.forEach(ch->{
            if(ch != channel){  //不是当前的channel，直接转发就行了。
                ch.writeAndFlush("【客户】"+channel.remoteAddress()+"发了一个消息是:"+msg+"\n");
            }else {//回显
                ch.writeAndFlush("[自己]发送了消息："+msg);
            }
        });
    }

    //当发生异常执行的方法
    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        //关闭通道就行了
        ctx.close();
    }
}
